package com.njbw.train.common.controller;

import com.njbw.train.common.exception.BusinessException;
import com.njbw.train.common.resp.CommonResp;
import lombok.extern.slf4j.Slf4j;
import org.springframework.validation.BindException;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.RestControllerAdvice;

/**
 * 统一异常处理、数据预处理等
 */
@Slf4j
@RestControllerAdvice
public class ControllerExceptionHandler {

    /**
     * 所有异常统一处理
     * @param e
     * @return
     */
    @ExceptionHandler(value = BusinessException.class)
    public CommonResp<?> exceptionHandler(BusinessException e) throws Exception {
        CommonResp<Object> commonResp = new CommonResp<>();
        log.error("业务异常：{}", e.getE().getDesc());
        commonResp.setSuccess(false);
        // commonResp.setMessage("系统出现异常，请联系管理员");
        commonResp.setMessage(e.getE().getDesc());
        return commonResp;
    }

    /**
     * 所有异常统一处理
     * @param e
     * @return
     */
    @ExceptionHandler(value = BindException.class)
    public CommonResp<?> businessExceptionHandler(BindException e) throws Exception {
        CommonResp<Object> commonResp = new CommonResp<>();
        // 这里只返回第一个错误，可以改成返回所有的校验错误
        String message = e.getBindingResult().getAllErrors().get(0).getDefaultMessage();
        log.error("参数校验异常：{}", message);
        commonResp.setSuccess(false);
        // commonResp.setMessage("系统出现异常，请联系管理员");
        commonResp.setMessage(message);
        return commonResp;
    }

    /**
     * 所有异常统一处理
     * @param e
     * @return
     */
    @ExceptionHandler(value = Exception.class)
    public CommonResp<?> exceptionHandler(Exception e) throws Exception {
        CommonResp<Object> commonResp = new CommonResp<>();
        log.error("系统异常", e);
        commonResp.setSuccess(false);
        commonResp.setMessage("系统出现异常，请联系管理员");
        return commonResp;
    }
}
